//package com.lucifer.cloud.boot.blog.ai.model.classification;
//
//import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;
//import org.deeplearning4j.optimize.listeners.ScoreIterationListener;
//import org.deeplearning4j.util.ModelSerializer;
//import org.nd4j.linalg.dataset.api.iterator.DataSetIterator;
//
///**
// * @author lucifer
// * @date 2024/11/28 13:52
// */
//public class TrainModel {
//    public static void main(String[] args) throws Exception {
//        // 数据加载
//        DataSetIterator trainIterator = ImageDataLoader.getDataSetIterator(Constant.trainImageDataPath);
//
//        // 创建并训练模型 4个类别
//        MultiLayerNetwork model = CNNModel.buildModel(Constant.numClasses);
//        // 每十个迭代打印一次损失函数值
//        model.setListeners(new ScoreIterationListener(10));
//
//        for (int i = 0; i < Constant.trainCount; i++) {
//            model.fit(trainIterator);
//            trainIterator.reset();
//        }
//
//        // 保存模型
//        ModelSerializer.writeModel(model, Constant.writeModelPath, true);
//    }
//}
